#include <iostream>
#include <queue>
#include <pthread.h>
#include <ctime>
#include <sys/types.h>
#include <unistd.h>
#include "BlockQueue.hpp"
#include "Task.hpp"
// 多生产者多消费者，没有S
// 阻塞队列结构体   C 计算任务  S 保存任务
template <class C, class S>
struct BlockQueues
{
    BlockQueue<C> *c_bq;
    BlockQueue<S> *s_bq;
};

// 生产者  参与生产
void *productor(void *args)
{
    BlockQueue<CalTask> *bq = (static_cast<BlockQueues<CalTask, SaveTask> *>(args))->c_bq;
    while (1)
    {
        // 生产数据
        int x = rand() % 10 + 1;
        int y = rand() % 5;
        int operIndex = rand() % oper.size();
        CalTask t(x, y, oper[operIndex], myMath);
        bq->push(t); // 把任务push进阻塞队列
        printf("Productor thread, 生产计算任务: %s\n", t.toString().c_str());
        sleep(1);
    }
    return nullptr;
}
// 消费者 参与计算和保存（计算的消费者，保存的生产者）
void *consumer(void *args)
{
    BlockQueue<CalTask> *bq = (static_cast<BlockQueues<CalTask, SaveTask> *>(args))->c_bq;       // 计算
    BlockQueue<SaveTask> *save_bq = (static_cast<BlockQueues<CalTask, SaveTask> *>(args))->s_bq; // 存储

    while (1)
    {
        // 消费数据
        CalTask t;
        bq->pop(&t);
    std:
        std::string result = t();
        printf("Cal thread, 完成计算任务:%s\n", result.c_str()); // operator()
        // sleep(1);
        // SaveTask save_task(result, Save);
        // save_bq->push(save_task);
        // printf("Cal thread, 推送存储任务完成\n");
    }
    return nullptr;
}
void *save(void *args)
{
    BlockQueue<SaveTask> *bq = (static_cast<BlockQueues<CalTask, SaveTask> *>(args))->s_bq;
    while (1)
    {
        SaveTask t;
        bq->pop(&t);
        t(); // 调用Save方法
        printf("Save thread, 保存任务完成\n");
    }
    return nullptr;
}
#define NUM 10
int main()
{
    srand((unsigned int)time(nullptr) ^ getpid());
    BlockQueues<CalTask, SaveTask> bqs;
    bqs.c_bq = new BlockQueue<CalTask>();
    bqs.s_bq = new BlockQueue<SaveTask>();
    pthread_t c[2], p[3];
    std::vector<pthread_t> consumers(NUM / 2), productors(NUM);
    for (auto& c : consumers) 
    {
        pthread_create(&c, nullptr, consumer, &bqs);
    }
    for (auto& p : productors)
    {
        pthread_create(&p, nullptr, productor, &bqs);
    }
    // pthread_create(&s, nullptr, save, &bqs);
    for (auto& c : consumers)
    {
        pthread_join(c, nullptr);
    }
    for (auto& p : productors)
    {
        pthread_join(p, nullptr);
    }

    // pthread_join(s, nullptr);
    delete bqs.c_bq;
    delete bqs.s_bq;
    return 0;
}